from january_scripts.ict4growth import ArtMAP


a = ArtMAP()
a.add_users([1, 2, 3, 4])
a.add_books(['a', 'b', 'c', 'd', 'e'])
a.add_scores([(1, 'a', 2), (1, 'b', 3), (2, 'b', 3), (2, 'c', 2), (3, 'c', 4), (4, 'a', 1)])

a.create_graph()

print a.bigraph.nodes()

# print a.book_density()
# print a.user_density()
#
# G = bipartite.projected_graph(a.bigraph, a.user_nodes)
# print G.nodes(), a.user_nodes
# print G.edges()
